python algorithm interview questions